package com.yiruantong.system.controller.monitor;

import cn.dev33.satoken.annotation.SaCheckPermission;
import cn.dev33.satoken.exception.NotLoginException;
import cn.dev33.satoken.stp.StpUtil;
import cn.hutool.core.bean.BeanUtil;
import com.yiruantong.system.domain.permission.SysUserOnline;
import lombok.RequiredArgsConstructor;
import com.yiruantong.common.core.constant.CacheConstants;
import com.yiruantong.common.core.domain.R;
import com.yiruantong.common.core.domain.dto.UserOnlineDTO;
import com.yiruantong.common.core.utils.StreamUtils;
import com.yiruantong.common.core.utils.StringUtils;
import com.yiruantong.common.log.annotation.Log;
import com.yiruantong.common.log.enums.BusinessType;
import com.yiruantong.common.mybatis.core.page.TableDataInfo;
import com.yiruantong.common.redis.utils.RedisUtils;
import com.yiruantong.common.web.core.BaseController;
import org.springframework.web.bind.annotation.*;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

/**
 * 在线用户监控
 *
 * @author YiRuanTong
 */
@RequiredArgsConstructor
@RestController
@RequestMapping("/monitor/online")
public class SysUserOnlineController extends BaseController {

  /**
   * 获取在线用户监控列表
   *
   * @param ipaddr   IP地址
   * @param userName 用户名
   */
  @SaCheckPermission("monitor:online:list")
  @GetMapping("/list")
  public TableDataInfo<SysUserOnline> list(String ipaddr, String userName) {
    // 获取所有未过期的 token
    List<String> keys = StpUtil.searchTokenValue("", 0, -1, false);
    List<UserOnlineDTO> userOnlineDTOList = new ArrayList<>();
    for (String key : keys) {
      String token = StringUtils.substringAfterLast(key, ":");
      // 如果已经过期则跳过
      if (StpUtil.stpLogic.getTokenActiveTimeoutByToken(token) < -1) {
        continue;
      }
      userOnlineDTOList.add(RedisUtils.getCacheObject(CacheConstants.ONLINE_TOKEN_KEY + token));
    }
    if (StringUtils.isNotEmpty(ipaddr) && StringUtils.isNotEmpty(userName)) {
      userOnlineDTOList = StreamUtils.filter(userOnlineDTOList, userOnline ->
        StringUtils.equals(ipaddr, userOnline.getIpaddr()) &&
          StringUtils.equals(userName, userOnline.getUserName())
      );
    } else if (StringUtils.isNotEmpty(ipaddr)) {
      userOnlineDTOList = StreamUtils.filter(userOnlineDTOList, userOnline ->
        StringUtils.equals(ipaddr, userOnline.getIpaddr())
      );
    } else if (StringUtils.isNotEmpty(userName)) {
      userOnlineDTOList = StreamUtils.filter(userOnlineDTOList, userOnline ->
        StringUtils.equals(userName, userOnline.getUserName())
      );
    }
    Collections.reverse(userOnlineDTOList);
    userOnlineDTOList.removeAll(Collections.singleton(null));
    List<SysUserOnline> userOnlineList = BeanUtil.copyToList(userOnlineDTOList, SysUserOnline.class);
    return TableDataInfo.build(userOnlineList);
  }

  /**
   * 强退用户
   *
   * @param tokenId token值
   */
  @SaCheckPermission("monitor:online:forceLogout")
  @Log(title = "在线用户", businessType = BusinessType.FORCE)
  @DeleteMapping("/{tokenId}")
  public R<Void> forceLogout(@PathVariable String tokenId) {
    try {
      StpUtil.kickoutByTokenValue(tokenId);
    } catch (NotLoginException ignored) {
    }
    return R.ok();
  }
}
